package com.article.interceptor;

import com.article.dto.ReturnError;
import com.article.enums.ReturnInfoEnum;
import com.article.model.MenuInfo;
import com.article.utils.ResponseUtil;
import net.sf.json.JSONObject;
import org.apache.log4j.Logger;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.Subject;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;

public class LoginInterceptor extends HandlerInterceptorAdapter {
    public static final String LAST_PAGE = "";
    private static final Logger log = Logger.getLogger(LoginInterceptor.class);// 日志文件

	/*
     * 利用正则映射到需要拦截的路径
	 * 
	 * private String mappingURL;
	 * 
	 * public void setMappingURL(String mappingURL) { this.mappingURL =
	 * mappingURL; }
	 */

    public static void Authority(String method, Integer type, String callback, HttpServletResponse response)
            throws Exception {
        JSONObject jsonObject = null;
        ReturnError returnError = new ReturnError();
        Subject currentUser = SecurityUtils.getSubject();
        Session session = currentUser.getSession();
        if ("GET".equalsIgnoreCase(method)) {
            if (type == 1) {
                session.removeAttribute("adminLogin");
                session.removeAttribute("Auths");
                session.removeAttribute("Menus");
                response.sendRedirect("../admin/login.html");
            } else {
                response.sendRedirect("../authenticationError.html");
            }
        } else {
            if (type == 1) {
                session.removeAttribute("adminLogin");
                session.removeAttribute("Auths");
                session.removeAttribute("Menus");
                returnError.setHasError(true);
                returnError.setError("会话过期，请重新登录");
                returnError.setErrUrl("login.html");
                jsonObject = JSONObject.fromObject(returnError);
                ResponseUtil.write(response, callback, jsonObject);
                //return new ReturnError(true, ReturnInfoEnum.NOSESSION.getType(), "会话过期，请重新登录",
                //        "login.html");
            } else {
                returnError.setHasError(true);
                returnError.setErrType(ReturnInfoEnum.NOAUTH.getType());
                returnError.setError(ReturnInfoEnum.NOAUTH.getInfo());
                jsonObject = JSONObject.fromObject(returnError);
                ResponseUtil.write(response, callback, jsonObject);
                //return new ReturnError(true, ReturnInfoEnum.NOAUTH.getType(), ReturnInfoEnum.NOAUTH.getInfo(), null);
            }
        }
    }

    /**
     * 在业务处理器处理请求之前被调用 如果返回false 从当前的拦截器往回执行所有拦截器的afterCompletion(),再退出拦截器链
     * 如果返回true 执行下一个拦截器,直到所有的拦截器都执行完毕 再执行被拦截的Controller 然后进入拦截器链,
     * 从最后一个拦截器往回执行所有的postHandle() 接着再从最后一个拦截器往回执行所有的afterCompletion()
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        // log.info("==============执行顺序: 1、preHandle================");
        // System.out.println("==============执行顺序:1、preHandle================");
        String requestUri = request.getRequestURI();
        String contextPath = request.getContextPath();
        String url = requestUri.substring(contextPath.length());
        // log.info("requestUri:" + requestUri);
        // log.info("contextPath:" + contextPath);
        // log.info("url:" + url);
        // System.out.println("method:" + request.getMethod());
        // System.out.println("requestUri:" + requestUri);
        // System.out.println("contextPath:" + contextPath);
        // System.out.println("url:" + url);
        Subject currentUser = SecurityUtils.getSubject();
        Session session = currentUser.getSession();
        String callback = request.getParameter("callback");
        // System.out.println(callback);
        if (true) {

            // return true;
        }
        if (session.getAttribute("adminLogin") == null || session.getAttribute("Auths") == null) {
            Authority(request.getMethod(), 1, callback, response);
            return false;
        }
        List<MenuInfo> auths = (List<MenuInfo>) session.getAttribute("Auths"); // 获取菜单列表
        // if (auths.size() <= 0) {
        // Authority(request.getMethod(), 2, callback, response);
        // return false;
        // }
        for (MenuInfo auth : auths) {
            // System.out.println(auth);
            if (auth.getMenuUrl().equals(url)) {
                // System.out.println(auth.isHasMenu());
                if (auth.isHasMenu()) {
                    return true;
                }
            }
        }
        Authority(request.getMethod(), 2, callback, response);
        return false;
    }

    /**
     * 在业务处理器处理请求执行完成后,生成视图之前执行的动作 可在modelAndView中加入数据，比如当前时间
     */
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
                           ModelAndView modelAndView) throws Exception {
        // log.info("==============执行顺序: 2、postHandle================");
        // System.out.println("==============执行顺序:
        // 2、postHandle================");
        if (modelAndView != null) { // 加入当前时间
            modelAndView.addObject("var", "测试postHandle");
        }
    }

    /**
     * 在DispatcherServlet完全处理完请求后被调用,可用于清理资源等
     * <p>
     * 当有拦截器抛出异常时,会从当前拦截器往回执行所有的拦截器的afterCompletion()
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
            throws Exception {
        // log.info("==============执行顺序: 3、afterCompletion================");
        // System.out
        // .println("==============执行顺序: 3、afterCompletion================");
    }
}
